HomeSort by relevance Sort by last modified time
    Searched defs:b_values (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
ctc_loss_op.cc 122 std::vector<int>* b_values = &labels_t[batch_indices]; variable
123 b_values->resize(values.size());
124 for (int i = 0; i < values.size(); ++i) (*b_values)[i] = values(i);
sparse_add_op.cc 48 OP_REQUIRES_OK(ctx, ctx->input("b_values", &b_values_t));
58 auto b_values = ctx->input(4).vec<T>(); variable
60 ctx, a_values.size() == a_nnz && b_values.size() == b_nnz,
63 a_values.size(), " and ", b_values.size()));
117 s = a_values(i) + b_values(j);
127 out_values.push_back(b_values(j));
sparse_sparse_binary_op_shared.cc 66 typename TTypes<T>::ConstFlat b_values, int64 b_nnz, int num_dims,
87 b_augmented_values->push_back(b_values(j));
94 b_augmented_values->push_back(b_values(j));
108 b_augmented_values->push_back(b_values(j++));
130 OP_REQUIRES_OK(ctx, ctx->input("b_values", &b_values_t));
146 "Inputs a_values and b_values should be vectors "
154 const auto b_values = b_values_t->vec<T>(); variable
157 ctx, a_values.size() == a_nnz && b_values.size() == b_nnz,
160 a_values.size(), " and ", b_values.size()));
189 b_values, b_nnz, num_dims, &a_augmented_values
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tests/
array_elementwise_ops_test.cc 316 std::vector<float> b_values; local
319 b_values.push_back(2 * i / static_cast<float>(count + 2));
328 Literal b_literal = LiteralUtil::CreateR1<float>({b_values});
332 auto b_constant = ConstantR1<float>(&builder, b_values);
345 expected.push_back(4 * (a_values[i] + b_values[i]));
    [all...]

Completed in 1398 milliseconds