Searched defs:is_batch (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 2296 bool is_batch, int num_examples, in ParseContextDenseFeatures()
2359 bool is_batch, int num_examples, in ParseContextSparseFeatures()
2421 bool is_batch, int num_examples, in ParseContextRaggedFeatures()
2499 bool is_batch, int num_examples, in ParseSequenceDenseFeatures()
2653 gtl::ArraySlice<tstring> example_names, bool is_batch, int num_examples, in ParseSequenceSparseFeatures()
2781 gtl::ArraySlice<tstring> example_names, bool is_batch, int num_examples, in ParseSequenceRaggedFeatures()
2932 bool is_batch) { in FastParseSequenceExample()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | matmul_utils.cc | 56 bool is_batch = absl::c_count(batch_dims, dim) != 0; in GetNonContractingDims() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad.cc | 1035 Status MatMulGradHelper(const Scope& scope, const bool is_batch, in MatMulGradHelper()
1062 const bool is_batch, in MatMulGradCommon()
|
| D | math_grad_test.cc | 606 void TestMatMulGrad(const bool is_batch, const bool t_x, const bool t_y) { in TestMatMulGrad()
631 void RandMatMulShapes(const bool is_batch, const bool tx, const bool ty, in RandMatMulShapes()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | example_parsing_ops.cc | 466 bool is_batch = TensorShapeUtils::IsVector(serialized->shape()); in Compute() local
|
| /external/libcxx/utils/google-benchmark/include/benchmark/ |
| D | benchmark.h | 694 bool is_batch) { in KeepRunningInternal()
|
| /external/google-benchmark/include/benchmark/ |
| D | benchmark.h | 701 bool is_batch) { in KeepRunningInternal()
|