Searched refs:sum_shape (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | aggregate_ops.cc | 40 TensorShape sum_shape; in Compile() local
42 GetTensorListBufferShape(ctx->Input(0), &sum_shape)); in Compile()
51 ctx, sum_shape.dim_sizes() == operand_shape.dim_sizes(), in Compile()
54 "shape.\n", "Expected: ", sum_shape.DebugString(), "\n", in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_conditional_accumulator.h | 258 TensorShape sum_shape = grad_val->shape(); in AddToAccumGradFunction() local
259 sum_shape.set_dim(0, sum_nnz); in AddToAccumGradFunction()
262 ctx, ctx->allocate_persistent(dtype_, sum_shape, tensor_sum_persistent, in AddToAccumGradFunction()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 188 auto sum_shape = builder.GetShape(sum).ConsumeValueOrDie(); in XLA_TEST_P() local
191 ShapeUtil::ElementsIn(sum_shape)); in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 31970 name: "sum_shape"
32050 name: "sum_shape"
|
| D | ops_history.v2.pbtxt | 71192 name: "sum_shape"
71274 name: "sum_shape"
71360 name: "sum_shape"
71448 name: "sum_shape"
|
| D | ops_history.v1.pbtxt | 71819 name: "sum_shape"
71901 name: "sum_shape"
71987 name: "sum_shape"
72075 name: "sum_shape"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 13991 …e tf.Output, thresh tf.Output) (sum_indices tf.Output, sum_values tf.Output, sum_shape tf.Output) {
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 35207 name: "sum_shape"
|